from transformers import MarianMTModel, MarianTokenizer
src = 'jap'  # source language
trg = 'en'  # target language
src_text = [
    '>>jap<< すべて の 事 は 人 を うみ疲れ させ る , 人 は これ を 言 いつく す こと が でき な い . \
    目 は 見 る こと に 飽き る こと が な く , 耳 は 聞 く こと に 満足 する こと が な い .',
]
model_name = f'D:/gitee_codes/transformer-learning/models/Helsinki-NLP/opus-mt-{src}-{trg}'
tokenizer = MarianTokenizer.from_pretrained(model_name)
model = MarianMTModel.from_pretrained(model_name)
translated = model.generate(**tokenizer.prepare_translation_batch(src_text))
print([tokenizer.decode(t, skip_special_tokens=True) for t in translated])